# Copyright (c) 2020 Nekokatt
# Copyright (c) 2021-present davfsa
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
from __future__ import annotations

import datetime

import mock
import pytest

from hikari import channels as channels_
from hikari import colors
from hikari import guilds
from hikari import permissions
from hikari import snowflakes
from hikari import undefined
from hikari import urls
from hikari import users
from hikari.impl import gateway_bot
from hikari.internal import routes
from hikari.internal import time
from tests.hikari import hikari_test_helpers


@pytest.fixture
def mock_app():
    return mock.Mock(spec_set=gateway_bot.GatewayBot)


class TestPartialRole:
    @pytest.fixture
    def model(self, mock_app):
        return guilds.PartialRole(app=mock_app, id=snowflakes.Snowflake(1106913972), name="The Big Cool")

    def test_str_operator(self, model):
        assert str(model) == "The Big Cool"

    def test_mention_property(self, model):
        assert model.mention == "<@&1106913972>"


def test_PartialApplication_str_operator():
    mock_application = mock.Mock(guilds.PartialApplication)
    mock_application.name = "beans"
    assert guilds.PartialApplication.__str__(mock_application) == "beans"


class TestPartialApplication:
    @pytest.fixture
    def model(self):
        return hikari_test_helpers.mock_class_namespace(
            guilds.PartialApplication, init_=False, slots_=False, id=123, icon_hash="ahashicon"
        )()

    def test_make_icon_url_format_set_to_deprecated_ext_argument_if_provided(self, model):
        with mock.patch.object(
            routes, "CDN_APPLICATION_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format="JPEG") == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, application_id=123, hash="ahashicon", size=4096, file_format="JPEG", lossless=True
        )

    def test_make_icon_url_when_hash_is_None(self, model):
        model.icon_hash = None

        with mock.patch.object(
            routes, "CDN_APPLICATION_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format="JPEG", size=1) is None

        route.compile_to_file.assert_not_called()

    def test_make_icon_url_when_hash_is_not_None(self, model):
        with mock.patch.object(
            routes, "CDN_APPLICATION_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format="JPEG", size=1) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, application_id=123, hash="ahashicon", size=1, file_format="JPEG", lossless=True
        )


class TestIntegrationAccount:
    @pytest.fixture
    def model(self, mock_app):
        return guilds.IntegrationAccount(id="foo", name="bar")

    def test_str_operator(self, model):
        assert str(model) == "bar"


class TestPartialIntegration:
    @pytest.fixture
    def model(self, mock_app):
        return guilds.PartialIntegration(
            account=mock.Mock(return_value=guilds.IntegrationAccount),
            id=snowflakes.Snowflake(69420),
            name="nice",
            type="illegal",
        )

    def test_str_operator(self, model):
        assert str(model) == "nice"


class TestRole:
    @pytest.fixture
    def model(self, mock_app):
        return guilds.Role(
            app=mock_app,
            id=snowflakes.Snowflake(979899100),
            name="@everyone",
            color=colors.Color(0x1A2B3C),
            guild_id=snowflakes.Snowflake(112233),
            is_hoisted=False,
            icon_hash="icon_hash",
            unicode_emoji=None,
            is_managed=False,
            is_mentionable=True,
            permissions=permissions.Permissions.CONNECT,
            position=12,
            bot_id=None,
            integration_id=None,
            is_premium_subscriber_role=False,
            is_guild_linked_role=True,
            subscription_listing_id=snowflakes.Snowflake(10),
            is_available_for_purchase=True,
        )

    def test_colour_property(self, model):
        assert model.colour == colors.Color(0x1A2B3C)

    def test_make_icon_url_format_set_to_deprecated_ext_argument_if_provided(self, model):
        with mock.patch.object(
            routes, "CDN_ROLE_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format="JPEG") == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, role_id=979899100, hash="icon_hash", size=4096, file_format="JPEG", lossless=True
        )

    def test_mention_property(self, model):
        assert model.mention == "<@&979899100>"

    def test_mention_property_when_is_everyone_role(self, model):
        model.id = model.guild_id
        assert model.mention == "@everyone"

    def test_make_icon_url_when_hash_is_None(self, model):
        model.icon_hash = None

        with mock.patch.object(
            routes, "CDN_ROLE_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format="JPEG", size=1) is None

        route.compile_to_file.assert_not_called()

    def test_make_icon_url_when_hash_is_not_None(self, model):
        with mock.patch.object(
            routes, "CDN_ROLE_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format="JPEG", size=1) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, role_id=979899100, hash="icon_hash", size=1, file_format="JPEG", lossless=True
        )


class TestGuildWidget:
    @pytest.fixture
    def model(self, mock_app):
        return guilds.GuildWidget(app=mock_app, channel_id=snowflakes.Snowflake(420), is_enabled=True)

    def test_app_property(self, model, mock_app):
        assert model.app is mock_app

    def test_channel_property(self, model):
        assert model.channel_id == snowflakes.Snowflake(420)

    def test_is_enabled_property(self, model):
        assert model.is_enabled is True

    @pytest.mark.asyncio
    async def test_fetch_channel(self, model):
        mock_channel = mock.Mock(channels_.GuildChannel)
        model.app.rest.fetch_channel = mock.AsyncMock(return_value=mock_channel)

        assert await model.fetch_channel() is model.app.rest.fetch_channel.return_value
        model.app.rest.fetch_channel.assert_awaited_once_with(420)

    @pytest.mark.asyncio
    async def test_fetch_channel_when_None(self, model):
        model.app.rest.fetch_channel = mock.AsyncMock()
        model.channel_id = None

        assert await model.fetch_channel() is None


class TestMember:
    @pytest.fixture
    def mock_user(self):
        return mock.Mock(id=snowflakes.Snowflake(123))

    @pytest.fixture
    def model(self, mock_user):
        return guilds.Member(
            guild_id=snowflakes.Snowflake(456),
            is_deaf=True,
            is_mute=True,
            is_pending=False,
            joined_at=datetime.datetime.now().astimezone(),
            nickname="davb",
            guild_avatar_decoration=users.AvatarDecoration(
                asset_hash="dimmadab",
                sku_id=snowflakes.Snowflake(123456789),
                expires_at=datetime.datetime(2025, 4, 4, 1, 1, 1),
            ),
            guild_avatar_hash="dab",
            guild_banner_hash="dimma",
            premium_since=None,
            role_ids=[snowflakes.Snowflake(456), snowflakes.Snowflake(1234)],
            user=mock_user,
            raw_communication_disabled_until=None,
            guild_flags=guilds.GuildMemberFlags.NONE,
        )

    def test_str_operator(self, model, mock_user):
        assert str(model) == str(mock_user)

    def test_app_property(self, model):
        assert model.app is model.user.app

    def test_id_property(self, model):
        assert model.id is model.user.id

    def test_username_property(self, model):
        assert model.username is model.user.username

    def test_discriminator_property(self, model):
        assert model.discriminator is model.user.discriminator

    def test_avatar_hash_property(self, model):
        assert model.avatar_hash is model.user.avatar_hash

    def test_is_bot_property(self, model):
        assert model.is_bot is model.user.is_bot

    def test_is_system_property(self, model):
        assert model.is_system is model.user.is_system

    def test_flags_property(self, model):
        assert model.flags is model.user.flags

    def test_display_avatar_decoration_property_when_guild_avatar_decoration_is_set(self, model):
        assert model.display_avatar_decoration is model.guild_avatar_decoration

    def test_display_avatar_decoration_property_when_guild_avatar_decoration_is_None(self, model):
        model.guild_avatar_decoration = None
        assert model.display_avatar_decoration is model.user.display_avatar_decoration

    def test_display_avatar_url_when_guild_hash_is_None(self, model):
        with mock.patch.object(guilds.Member, "make_guild_avatar_url") as mock_make_guild_avatar_url:
            assert model.display_avatar_url is mock_make_guild_avatar_url.return_value

    def test_display_guild_avatar_url_when_guild_hash_is_not_None(self, model):
        with mock.patch.object(guilds.Member, "make_guild_avatar_url", return_value=None):
            assert model.display_avatar_url is model.user.display_avatar_url

    def test_display_banner_url_when_guild_hash_is_None(self, model):
        with mock.patch.object(guilds.Member, "make_guild_banner_url") as mock_make_guild_banner_url:
            assert model.display_banner_url is mock_make_guild_banner_url.return_value

    def test_display_banner_url_when_guild_hash_is_not_None(self, model):
        with mock.patch.object(guilds.Member, "make_guild_banner_url", return_value=None):
            assert model.display_banner_url is model.user.display_banner_url

    def test_banner_hash_property(self, model):
        assert model.banner_hash is model.user.banner_hash

    def test_accent_color_property(self, model):
        assert model.accent_color is model.user.accent_color

    def test_communication_disabled_until(self, model):
        model.raw_communication_disabled_until = datetime.datetime(2021, 11, 22)

        with mock.patch.object(time, "utc_datetime", return_value=datetime.datetime(2021, 10, 18)):
            assert model.communication_disabled_until() == datetime.datetime(2021, 11, 22)

    def test_communication_disabled_until_when_raw_communication_disabled_until_is_None(self, model):
        model.raw_communication_disabled_until = None

        with mock.patch.object(time, "utc_datetime", return_value=datetime.datetime(2021, 10, 18)):
            assert model.communication_disabled_until() is None

    def test_communication_disabled_until_when_raw_communication_disabled_until_is_in_the_past(self, model):
        model.raw_communication_disabled_until = datetime.datetime(2021, 10, 18)

        with mock.patch.object(time, "utc_datetime", return_value=datetime.datetime(2021, 11, 22)):
            assert model.communication_disabled_until() is None

    def test_make_avatar_url_format_set_to_deprecated_ext_argument_if_provided(self, model):
        with mock.patch.object(
            routes, "CDN_MEMBER_AVATAR", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_guild_avatar_url(file_format="JPEG") == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=456, user_id=123, hash="dab", size=4096, file_format="JPEG", lossless=True
        )

    def test_make_avatar_url(self, model):
        result = model.make_avatar_url(file_format="PNG", size=4096)

        model.user.make_avatar_url.assert_called_once_with(file_format="PNG", size=4096, lossless=True)
        assert result is model.user.make_avatar_url.return_value

    def test_make_guild_avatar_url_when_no_hash(self, model):
        model.guild_avatar_hash = None
        assert model.make_guild_avatar_url(file_format="PNG", size=1024) is None

    def test_make_guild_avatar_url_when_format_is_None_and_avatar_hash_is_for_animated(self, model):
        model.guild_avatar_hash = "a_18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_MEMBER_AVATAR", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_guild_avatar_url(file_format=None, size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL,
            guild_id=model.guild_id,
            user_id=model.id,
            hash=model.guild_avatar_hash,
            size=4096,
            file_format="GIF",
            lossless=True,
        )

    def test_make_guild_avatar_url_when_format_is_None_and_avatar_hash_is_not_for_animated(self, model):
        model.guild_avatar_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_MEMBER_AVATAR", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_guild_avatar_url(file_format=None, size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL,
            guild_id=model.guild_id,
            user_id=model.id,
            hash=model.guild_avatar_hash,
            size=4096,
            file_format="PNG",
            lossless=True,
        )

    def test_make_guild_avatar_url_with_all_args(self, model):
        model.guild_avatar_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_MEMBER_AVATAR", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_guild_avatar_url(file_format="URL", size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL,
            guild_id=model.guild_id,
            user_id=model.id,
            hash=model.guild_avatar_hash,
            size=4096,
            file_format="URL",
            lossless=True,
        )

    def test_make_banner_url_format_set_to_deprecated_ext_argument_if_provided(self, model):
        with mock.patch.object(
            routes, "CDN_MEMBER_BANNER", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_guild_banner_url(file_format="JPEG") == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL,
            user_id=model.id,
            guild_id=model.guild_id,
            hash=model.guild_banner_hash,
            size=4096,
            file_format="JPEG",
            lossless=True,
        )

    def test_make_banner_url(self, model):
        result = model.make_banner_url(file_format="PNG", size=4096)

        model.user.make_banner_url.assert_called_once_with(
            file_format="PNG", size=4096, lossless=True
        )
        assert result is model.user.make_banner_url.return_value

    def test_make_guild_banner_url_when_no_hash(self, model):
        model.guild_banner_hash = None
        assert model.make_guild_banner_url(file_format="PNG", size=1024) is None

    def test_make_guild_banner_url_when_format_is_None_and_banner_hash_is_for_animated(self, model):
        model.guild_banner_hash = "a_18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_MEMBER_BANNER", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_guild_banner_url(file_format=None, size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL,
            user_id=model.id,
            guild_id=model.guild_id,
            hash=model.guild_banner_hash,
            size=4096,
            file_format="GIF",
            lossless=True,
        )

    def test_make_guild_banner_url_when_format_is_None_and_banner_hash_is_not_for_animated(self, model):
        model.guild_banner_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_MEMBER_BANNER", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_guild_banner_url(file_format=None, size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL,
            user_id=model.id,
            guild_id=model.guild_id,
            hash=model.guild_banner_hash,
            size=4096,
            file_format="PNG",
            lossless=True,
        )

    def test_make_guild_banner_url_with_all_args(self, model):
        model.guild_banner_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_MEMBER_BANNER", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_guild_banner_url(file_format="URL", size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL,
            guild_id=model.guild_id,
            user_id=model.id,
            hash=model.guild_banner_hash,
            size=4096,
            file_format="URL",
            lossless=True,
        )

    @pytest.mark.asyncio
    async def test_fetch_dm_channel(self, model):
        model.user.fetch_dm_channel = mock.AsyncMock()

        assert await model.fetch_dm_channel() is model.user.fetch_dm_channel.return_value

        model.user.fetch_dm_channel.assert_awaited_once_with()

    @pytest.mark.asyncio
    async def test_fetch_self(self, model):
        model.user.app.rest.fetch_member = mock.AsyncMock()

        assert await model.fetch_self() is model.user.app.rest.fetch_member.return_value

        model.user.app.rest.fetch_member.assert_awaited_once_with(456, 123)

    @pytest.mark.asyncio
    async def test_fetch_roles(self, model):
        model.user.app.rest.fetch_roles = mock.AsyncMock()
        await model.fetch_roles()
        model.user.app.rest.fetch_roles.assert_awaited_once_with(456)

    @pytest.mark.asyncio
    async def test_ban(self, model):
        model.app.rest.ban_user = mock.AsyncMock()

        await model.ban(delete_message_seconds=600, reason="bored")

        model.app.rest.ban_user.assert_awaited_once_with(456, 123, delete_message_seconds=600, reason="bored")

    @pytest.mark.asyncio
    async def test_unban(self, model):
        model.app.rest.unban_user = mock.AsyncMock()

        await model.unban(reason="Unbored")

        model.app.rest.unban_user.assert_awaited_once_with(456, 123, reason="Unbored")

    @pytest.mark.asyncio
    async def test_kick(self, model):
        model.app.rest.kick_user = mock.AsyncMock()

        await model.kick(reason="bored")

        model.app.rest.kick_user.assert_awaited_once_with(456, 123, reason="bored")

    @pytest.mark.asyncio
    async def test_add_role(self, model):
        model.app.rest.add_role_to_member = mock.AsyncMock()

        await model.add_role(563412, reason="Promoted")

        model.app.rest.add_role_to_member.assert_awaited_once_with(456, 123, 563412, reason="Promoted")

    @pytest.mark.asyncio
    async def test_remove_role(self, model):
        model.app.rest.remove_role_from_member = mock.AsyncMock()

        await model.remove_role(563412, reason="Demoted")

        model.app.rest.remove_role_from_member.assert_awaited_once_with(456, 123, 563412, reason="Demoted")

    @pytest.mark.asyncio
    async def test_edit(self, model):
        model.app.rest.edit_member = mock.AsyncMock()
        disabled_until = datetime.datetime(2021, 11, 17)
        edit = await model.edit(
            nickname="Imposter",
            roles=[123, 432, 345],
            mute=False,
            deaf=True,
            voice_channel=4321245,
            communication_disabled_until=disabled_until,
            reason="I'm God",
        )

        model.app.rest.edit_member.assert_awaited_once_with(
            456,
            123,
            nickname="Imposter",
            roles=[123, 432, 345],
            mute=False,
            deaf=True,
            voice_channel=4321245,
            communication_disabled_until=disabled_until,
            reason="I'm God",
        )

        assert edit == model.app.rest.edit_member.return_value

    def test_default_avatar_url_property(self, model):
        assert model.default_avatar_url is model.user.default_avatar_url

    def test_display_name_property_when_nickname(self, model):
        assert model.display_name == "davb"

    def test_display_name_property_when_no_nickname(self, model):
        model.nickname = None
        assert model.display_name is model.user.global_name

    def test_mention_property(self, model):
        assert model.mention == model.user.mention

    def test_get_guild(self, model):
        guild = mock.Mock(id=456)
        model.user.app.cache.get_guild.side_effect = [guild]

        assert model.get_guild() == guild

        model.user.app.cache.get_guild.assert_has_calls([mock.call(456)])

    def test_get_guild_when_guild_not_in_cache(self, model):
        model.user.app.cache.get_guild.side_effect = [None]

        assert model.get_guild() is None

        model.user.app.cache.get_guild.assert_has_calls([mock.call(456)])

    def test_get_guild_when_no_cache_trait(self, model):
        model.user.app = object()

        assert model.get_guild() is None

    def test_get_roles(self, model):
        role1 = mock.Mock(id=321, position=2)
        role2 = mock.Mock(id=654, position=1)
        model.user.app.cache.get_role.side_effect = [role1, role2]
        model.role_ids = [321, 654]

        assert model.get_roles() == [role1, role2]

        model.user.app.cache.get_role.assert_has_calls([mock.call(321), mock.call(654)])

    def test_get_roles_when_role_ids_not_in_cache(self, model):
        role = mock.Mock(id=456, position=1)
        model.user.app.cache.get_role.side_effect = [None, role]
        model.role_ids = [321, 456]

        assert model.get_roles() == [role]

        model.user.app.cache.get_role.assert_has_calls([mock.call(321), mock.call(456)])

    def test_get_roles_when_empty_cache(self, model):
        model.role_ids = [132, 432]
        model.user.app.cache.get_role.side_effect = [None, None]

        assert model.get_roles() == []

        model.user.app.cache.get_role.assert_has_calls([mock.call(132), mock.call(432)])

    def test_get_roles_when_no_cache_trait(self, model):
        model.user.app = object()

        assert model.get_roles() == []

    def test_get_top_role(self, model):
        role1 = mock.Mock(id=321, position=2)
        role2 = mock.Mock(id=654, position=1)

        with mock.patch.object(guilds.Member, "get_roles", return_value=[role1, role2]):
            assert model.get_top_role() is role1

    def test_get_top_role_when_roles_is_empty(self, model):
        with mock.patch.object(guilds.Member, "get_roles", return_value=[]):
            assert model.get_top_role() is None

    def test_get_presence(self, model):
        assert model.get_presence() is model.user.app.cache.get_presence.return_value
        model.user.app.cache.get_presence.assert_called_once_with(456, 123)

    def test_get_presence_when_no_cache_trait(self, model):
        model.user.app = object()
        assert model.get_presence() is None


class TestPartialGuild:
    @pytest.fixture
    def model(self, mock_app):
        return guilds.PartialGuild(app=mock_app, id=snowflakes.Snowflake(90210), icon_hash="yeet", name="hikari")

    def test_str_operator(self, model):
        assert str(model) == "hikari"

    def test_shard_id_property(self, model):
        model.app.shard_count = 4
        assert model.shard_id == 0

    def test_shard_id_when_not_shard_aware(self, model):
        model.app = object()

        assert model.shard_id is None

    def test_make_icon_url_format_set_to_deprecated_ext_argument_if_provided(self, model):
        with mock.patch.object(
            routes, "CDN_GUILD_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format="JPEG") == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=90210, hash="yeet", size=4096, file_format="JPEG", lossless=True
        )

    def test_make_icon_url_when_no_hash(self, model):
        model.icon_hash = None

        assert model.make_icon_url(file_format="PNG", size=2048) is None

    def test_make_icon_url_when_format_is_None_and_avatar_hash_is_for_animated(self, model):
        model.icon_hash = "a_yeet"

        with mock.patch.object(
            routes, "CDN_GUILD_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format=None, size=1024) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=90210, hash="a_yeet", size=1024, file_format="GIF", lossless=True
        )

    def test_make_icon_url_when_format_is_None_and_avatar_hash_is_not_for_animated(self, model):
        with mock.patch.object(
            routes, "CDN_GUILD_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format=None, size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=90210, hash="yeet", size=4096, file_format="PNG", lossless=True
        )

    def test_make_icon_url_with_all_args(self, model):
        with mock.patch.object(
            routes, "CDN_GUILD_ICON", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_icon_url(file_format="URL", size=2048) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=90210, hash="yeet", size=2048, file_format="URL", lossless=True
        )

    @pytest.mark.asyncio
    async def test_kick(self, model):
        model.app.rest.kick_user = mock.AsyncMock()
        await model.kick(4321, reason="Go away!")

        model.app.rest.kick_user.assert_awaited_once_with(90210, 4321, reason="Go away!")

    @pytest.mark.asyncio
    async def test_ban(self, model):
        model.app.rest.ban_user = mock.AsyncMock()

        await model.ban(4321, delete_message_seconds=864000, reason="Go away!")

        model.app.rest.ban_user.assert_awaited_once_with(90210, 4321, delete_message_seconds=864000, reason="Go away!")

    @pytest.mark.asyncio
    async def test_unban(self, model):
        model.app.rest.unban_user = mock.AsyncMock()
        await model.unban(4321, reason="Comeback!!")

        model.app.rest.unban_user.assert_awaited_once_with(90210, 4321, reason="Comeback!!")

    @pytest.mark.asyncio
    async def test_edit(self, model):
        model.app.rest.edit_guild = mock.AsyncMock()
        edited_guild = await model.edit(
            name="chad server",
            verification_level=guilds.GuildVerificationLevel.LOW,
            default_message_notifications=guilds.GuildMessageNotificationsLevel.ALL_MESSAGES,
            explicit_content_filter_level=guilds.GuildExplicitContentFilterLevel.DISABLED,
            owner=6996,
            afk_timeout=400,
            preferred_locale="us-en",
            features=[guilds.GuildFeature.COMMUNITY, guilds.GuildFeature.RAID_ALERTS_DISABLED],
            reason="beep boop",
        )

        model.app.rest.edit_guild.assert_awaited_once_with(
            90210,
            name="chad server",
            verification_level=guilds.GuildVerificationLevel.LOW,
            default_message_notifications=guilds.GuildMessageNotificationsLevel.ALL_MESSAGES,
            explicit_content_filter_level=guilds.GuildExplicitContentFilterLevel.DISABLED,
            afk_channel=undefined.UNDEFINED,
            afk_timeout=400,
            icon=undefined.UNDEFINED,
            owner=6996,
            splash=undefined.UNDEFINED,
            banner=undefined.UNDEFINED,
            system_channel=undefined.UNDEFINED,
            rules_channel=undefined.UNDEFINED,
            public_updates_channel=undefined.UNDEFINED,
            preferred_locale="us-en",
            features=[guilds.GuildFeature.COMMUNITY, guilds.GuildFeature.RAID_ALERTS_DISABLED],
            reason="beep boop",
        )

        assert edited_guild is model.app.rest.edit_guild.return_value

    @pytest.mark.asyncio
    async def test_set_incident_actions(self, model: guilds.PartialGuild):
        model.app.rest.set_guild_incident_actions = mock.AsyncMock()

        updated_incident_data = await model.set_incident_actions(
            invites_disabled_until=datetime.datetime(2021, 11, 17), dms_disabled_until=datetime.datetime(2021, 11, 18)
        )

        assert updated_incident_data is model.app.rest.set_guild_incident_actions.return_value
        model.app.rest.set_guild_incident_actions.assert_awaited_once_with(
            90210,
            invites_disabled_until=datetime.datetime(2021, 11, 17),
            dms_disabled_until=datetime.datetime(2021, 11, 18),
        )

    @pytest.mark.asyncio
    async def test_fetch_emojis(self, model):
        model.app.rest.fetch_guild_emojis = mock.AsyncMock()

        emojis = await model.fetch_emojis()

        model.app.rest.fetch_guild_emojis.assert_awaited_once_with(model.id)
        assert emojis is model.app.rest.fetch_guild_emojis.return_value

    @pytest.mark.asyncio
    async def test_fetch_emoji(self, model):
        model.app.rest.fetch_emoji = mock.AsyncMock()

        emoji = await model.fetch_emoji(349)

        model.app.rest.fetch_emoji.assert_awaited_once_with(model.id, 349)
        assert emoji is model.app.rest.fetch_emoji.return_value

    @pytest.mark.asyncio
    async def test_fetch_stickers(self, model):
        model.app.rest.fetch_guild_stickers = mock.AsyncMock()

        stickers = await model.fetch_stickers()

        model.app.rest.fetch_guild_stickers.assert_awaited_once_with(model.id)
        assert stickers is model.app.rest.fetch_guild_stickers.return_value

    @pytest.mark.asyncio
    async def test_fetch_sticker(self, model):
        model.app.rest.fetch_guild_sticker = mock.AsyncMock()

        sticker = await model.fetch_sticker(6969)

        model.app.rest.fetch_guild_sticker.assert_awaited_once_with(model.id, 6969)
        assert sticker is model.app.rest.fetch_guild_sticker.return_value

    @pytest.mark.asyncio
    async def test_create_sticker(self, model):
        model.app.rest.create_sticker = mock.AsyncMock()
        file = object()

        sticker = await model.create_sticker(
            "NewSticker", "funny", file, description="A sticker", reason="blah blah blah"
        )
        assert sticker is model.app.rest.create_sticker.return_value

        model.app.rest.create_sticker.assert_awaited_once_with(
            90210, "NewSticker", "funny", file, description="A sticker", reason="blah blah blah"
        )

    @pytest.mark.asyncio
    async def test_edit_sticker(self, model):
        model.app.rest.edit_sticker = mock.AsyncMock()

        sticker = await model.edit_sticker(4567, name="Brilliant", tag="parmesan", description="amazing")

        model.app.rest.edit_sticker.assert_awaited_once_with(
            90210, 4567, name="Brilliant", tag="parmesan", description="amazing", reason=undefined.UNDEFINED
        )

        assert sticker is model.app.rest.edit_sticker.return_value

    @pytest.mark.asyncio
    async def test_delete_sticker(self, model):
        model.app.rest.delete_sticker = mock.AsyncMock()

        sticker = await model.delete_sticker(951)

        model.app.rest.delete_sticker.assert_awaited_once_with(90210, 951, reason=undefined.UNDEFINED)

        assert sticker is model.app.rest.delete_sticker.return_value

    @pytest.mark.asyncio
    async def test_create_category(self, model):
        model.app.rest.create_guild_category = mock.AsyncMock()

        category = await model.create_category("very cool category", position=2)

        model.app.rest.create_guild_category.assert_awaited_once_with(
            90210,
            "very cool category",
            position=2,
            permission_overwrites=undefined.UNDEFINED,
            reason=undefined.UNDEFINED,
        )

        assert category is model.app.rest.create_guild_category.return_value

    @pytest.mark.asyncio
    async def test_create_text_channel(self, model):
        model.app.rest.create_guild_text_channel = mock.AsyncMock()

        text_channel = await model.create_text_channel(
            "cool text channel", position=3, nsfw=False, rate_limit_per_user=30
        )

        model.app.rest.create_guild_text_channel.assert_awaited_once_with(
            90210,
            "cool text channel",
            position=3,
            topic=undefined.UNDEFINED,
            nsfw=False,
            rate_limit_per_user=30,
            permission_overwrites=undefined.UNDEFINED,
            category=undefined.UNDEFINED,
            reason=undefined.UNDEFINED,
        )

        assert text_channel is model.app.rest.create_guild_text_channel.return_value

    @pytest.mark.asyncio
    async def test_create_news_channel(self, model):
        model.app.rest.create_guild_news_channel = mock.AsyncMock()

        news_channel = await model.create_news_channel(
            "cool news channel", position=1, nsfw=False, rate_limit_per_user=420
        )

        model.app.rest.create_guild_news_channel.assert_awaited_once_with(
            90210,
            "cool news channel",
            position=1,
            topic=undefined.UNDEFINED,
            nsfw=False,
            rate_limit_per_user=420,
            permission_overwrites=undefined.UNDEFINED,
            category=undefined.UNDEFINED,
            reason=undefined.UNDEFINED,
        )

        assert news_channel is model.app.rest.create_guild_news_channel.return_value

    @pytest.mark.asyncio
    async def test_create_forum_channel(self, model):
        model.app.rest.create_guild_forum_channel = mock.AsyncMock()

        forum_channel = await model.create_forum_channel(
            "cool forum channel", position=1, nsfw=False, rate_limit_per_user=420
        )

        model.app.rest.create_guild_forum_channel.assert_awaited_once_with(
            90210,
            "cool forum channel",
            position=1,
            topic=undefined.UNDEFINED,
            nsfw=False,
            rate_limit_per_user=420,
            permission_overwrites=undefined.UNDEFINED,
            category=undefined.UNDEFINED,
            reason=undefined.UNDEFINED,
            default_auto_archive_duration=undefined.UNDEFINED,
            default_thread_rate_limit_per_user=undefined.UNDEFINED,
            default_forum_layout=undefined.UNDEFINED,
            default_sort_order=undefined.UNDEFINED,
            available_tags=undefined.UNDEFINED,
            default_reaction_emoji=undefined.UNDEFINED,
        )

        assert forum_channel is model.app.rest.create_guild_forum_channel.return_value

    @pytest.mark.asyncio
    async def test_create_voice_channel(self, model):
        model.app.rest.create_guild_voice_channel = mock.AsyncMock()

        voice_channel = await model.create_voice_channel(
            "cool voice channel", position=1, bitrate=3200, video_quality_mode=2
        )

        model.app.rest.create_guild_voice_channel.assert_awaited_once_with(
            90210,
            "cool voice channel",
            position=1,
            user_limit=undefined.UNDEFINED,
            bitrate=3200,
            video_quality_mode=2,
            permission_overwrites=undefined.UNDEFINED,
            region=undefined.UNDEFINED,
            category=undefined.UNDEFINED,
            reason=undefined.UNDEFINED,
        )

        assert voice_channel is model.app.rest.create_guild_voice_channel.return_value

    @pytest.mark.asyncio
    async def test_create_stage_channel(self, model):
        model.app.rest.create_guild_stage_channel = mock.AsyncMock()

        stage_channel = await model.create_stage_channel("cool stage channel", position=1, bitrate=3200, user_limit=100)

        model.app.rest.create_guild_stage_channel.assert_awaited_once_with(
            90210,
            "cool stage channel",
            position=1,
            user_limit=100,
            bitrate=3200,
            permission_overwrites=undefined.UNDEFINED,
            region=undefined.UNDEFINED,
            category=undefined.UNDEFINED,
            reason=undefined.UNDEFINED,
        )

        assert stage_channel is model.app.rest.create_guild_stage_channel.return_value

    @pytest.mark.asyncio
    async def test_delete_channel(self, model):
        mock_channel = mock.Mock(channels_.GuildChannel)
        model.app.rest.delete_channel = mock.AsyncMock(return_value=mock_channel)

        deleted_channel = await model.delete_channel(1288820)

        model.app.rest.delete_channel.assert_awaited_once_with(1288820)
        assert deleted_channel is model.app.rest.delete_channel.return_value

    @pytest.mark.asyncio
    async def test_fetch_guild(self, model):
        model.app.rest.fetch_guild = mock.AsyncMock(return_value=model)

        assert await model.fetch_self() is model.app.rest.fetch_guild.return_value
        model.app.rest.fetch_guild.assert_awaited_once_with(model.id)

    @pytest.mark.asyncio
    async def test_fetch_roles(self, model):
        model.app.rest.fetch_roles = mock.AsyncMock()

        roles = await model.fetch_roles()

        model.app.rest.fetch_roles.assert_awaited_once_with(90210)
        assert roles is model.app.rest.fetch_roles.return_value


class TestGuildPreview:
    @pytest.fixture
    def model(self, mock_app):
        return guilds.GuildPreview(
            app=mock_app,
            features=["huge super secret nsfw channel"],
            id=snowflakes.Snowflake(123),
            icon_hash="dis is mah icon hash",
            name="DAPI",
            splash_hash="dis is also mah splash hash",
            discovery_splash_hash=None,
            emojis={},
            approximate_active_member_count=12,
            approximate_member_count=999_283_252_124_633,
            description="the place for quality shitposting!",
        )

    def test_make_splash_url_format_set_to_deprecated_ext_argument_if_provided(self, model):
        with mock.patch.object(
            routes, "CDN_GUILD_SPLASH", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_splash_url(file_format="JPEG") == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=123, hash="dis is also mah splash hash", size=4096, file_format="JPEG", lossless=True
        )

    def test_make_discovery_splash_url_format_set_to_deprecated_ext_argument_if_provided(self, model):
        model.discovery_splash_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_GUILD_DISCOVERY_SPLASH", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_discovery_splash_url(file_format="JPEG") == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=123, hash="18dnf8dfbakfdh", size=4096, file_format="JPEG", lossless=True
        )

    def test_make_splash_url_when_hash(self, model):
        model.splash_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_GUILD_SPLASH", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_splash_url(file_format="URL", size=1024) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=123, hash="18dnf8dfbakfdh", size=1024, file_format="URL", lossless=True
        )

    def test_make_splash_url_when_no_hash(self, model):
        model.splash_hash = None
        assert model.make_splash_url(file_format="png", size=512) is None

    def test_make_discovery_splash_url_when_hash(self, model):
        model.discovery_splash_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_GUILD_DISCOVERY_SPLASH", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_discovery_splash_url(file_format="URL", size=2048) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=123, hash="18dnf8dfbakfdh", size=2048, file_format="URL", lossless=True
        )

    def test_make_discovery_splash_url_when_no_hash(self, model):
        model.discovery_splash_hash = None
        assert model.make_discovery_splash_url(file_format="PNG", size=4096) is None


class TestGuild:
    @pytest.fixture
    def model(self, mock_app):
        return hikari_test_helpers.mock_class_namespace(guilds.Guild)(
            app=mock_app,
            id=snowflakes.Snowflake(123),
            splash_hash="splash_hash",
            discovery_splash_hash="discovery_splash_hash",
            banner_hash="banner_hash",
            icon_hash="icon_hash",
            features=[guilds.GuildFeature.ANIMATED_ICON],
            incidents=guilds.GuildIncidents(
                invites_disabled_until=None,
                dms_disabled_until=None,
                dm_spam_detected_at=datetime.datetime(2015, 5, 13, 1, 1, 1, 1, tzinfo=datetime.timezone.utc),
                raid_detected_at=None,
            ),
            name="some guild",
            application_id=snowflakes.Snowflake(9876),
            afk_channel_id=snowflakes.Snowflake(1234),
            afk_timeout=datetime.timedelta(seconds=60),
            default_message_notifications=guilds.GuildMessageNotificationsLevel.ONLY_MENTIONS,
            description=None,
            explicit_content_filter=guilds.GuildExplicitContentFilterLevel.ALL_MEMBERS,
            is_widget_enabled=False,
            max_video_channel_users=10,
            mfa_level=guilds.GuildMFALevel.NONE,
            nsfw_level=guilds.GuildNSFWLevel.AGE_RESTRICTED,
            owner_id=snowflakes.Snowflake(1111),
            preferred_locale="en-GB",
            premium_subscription_count=12,
            premium_tier=guilds.GuildPremiumTier.TIER_3,
            public_updates_channel_id=snowflakes.Snowflake(99699),
            rules_channel_id=snowflakes.Snowflake(123445),
            system_channel_id=snowflakes.Snowflake(123888),
            vanity_url_code="yeet",
            verification_level=guilds.GuildVerificationLevel.VERY_HIGH,
            widget_channel_id=snowflakes.Snowflake(192729),
            system_channel_flags=guilds.GuildSystemChannelFlag.SUPPRESS_PREMIUM_SUBSCRIPTION,
        )

    def test_get_channels(self, model):
        assert model.get_channels() is model.app.cache.get_guild_channels_view_for_guild.return_value
        model.app.cache.get_guild_channels_view_for_guild.assert_called_once_with(123)

    def test_get_channels_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_channels() == {}

    def test_get_members(self, model):
        assert model.get_members() is model.app.cache.get_members_view_for_guild.return_value
        model.app.cache.get_members_view_for_guild.assert_called_once_with(123)

    def test_get_members_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_members() == {}

    def test_get_presences(self, model):
        assert model.get_presences() is model.app.cache.get_presences_view_for_guild.return_value
        model.app.cache.get_presences_view_for_guild.assert_called_once_with(123)

    def test_get_presences_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_presences() == {}

    def test_get_voice_states(self, model):
        assert model.get_voice_states() is model.app.cache.get_voice_states_view_for_guild.return_value
        model.app.cache.get_voice_states_view_for_guild.assert_called_once_with(123)

    def test_get_voice_states_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_voice_states() == {}

    def test_get_emojis(self, model):
        assert model.get_emojis() is model.app.cache.get_emojis_view_for_guild.return_value
        model.app.cache.get_emojis_view_for_guild.assert_called_once_with(123)

    def test_emojis_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_emojis() == {}

    def test_get_sticker(self, model):
        model.app.cache.get_sticker.return_value.guild_id = model.id
        assert model.get_sticker(456) is model.app.cache.get_sticker.return_value
        model.app.cache.get_sticker.assert_called_once_with(456)

    def test_get_sticker_when_not_from_guild(self, model):
        model.app.cache.get_sticker.return_value.guild_id = 546123123433

        assert model.get_sticker(456) is None

        model.app.cache.get_sticker.assert_called_once_with(456)

    def test_get_sticker_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_sticker(1234) is None

    def test_get_stickers(self, model):
        assert model.get_stickers() is model.app.cache.get_stickers_view_for_guild.return_value
        model.app.cache.get_stickers_view_for_guild.assert_called_once_with(123)

    def test_get_stickers_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_stickers() == {}

    def test_roles(self, model):
        assert model.get_roles() is model.app.cache.get_roles_view_for_guild.return_value
        model.app.cache.get_roles_view_for_guild.assert_called_once_with(123)

    def test_get_roles_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_roles() == {}

    def test_get_emoji(self, model):
        model.app.cache.get_emoji.return_value.guild_id = model.id
        assert model.get_emoji(456) is model.app.cache.get_emoji.return_value
        model.app.cache.get_emoji.assert_called_once_with(456)

    def test_get_emoji_when_not_from_guild(self, model):
        model.app.cache.get_emoji.return_value.guild_id = 1233212

        assert model.get_emoji(456) is None

        model.app.cache.get_emoji.assert_called_once_with(456)

    def test_get_emoji_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_emoji(456) is None

    def test_get_role(self, model):
        model.app.cache.get_role.return_value.guild_id = model.id
        assert model.get_role(456) is model.app.cache.get_role.return_value
        model.app.cache.get_role.assert_called_once_with(456)

    def test_get_role_when_not_from_guild(self, model):
        model.app.cache.get_role.return_value.guild_id = 7623123321123

        assert model.get_role(456) is None

        model.app.cache.get_role.assert_called_once_with(456)

    def test_get_role_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_role(456) is None

    @pytest.mark.asyncio
    async def test_invites_disabled_default(self, model):
        assert model.invites_disabled is False

    @pytest.mark.asyncio
    async def test_invites_disabled_via_incidents(self, model):
        model.incidents = guilds.GuildIncidents(
            invites_disabled_until=datetime.datetime(2021, 11, 17),
            dms_disabled_until=None,
            dm_spam_detected_at=None,
            raid_detected_at=None,
        )
        assert model.invites_disabled is True

    @pytest.mark.asyncio
    async def test_invites_disabled_via_feature(self, model):
        model.features.append(guilds.GuildFeature.INVITES_DISABLED)
        assert model.invites_disabled is True

    def test_make_splash_url_when_hash(self, model):
        model.splash_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_GUILD_SPLASH", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_splash_url(file_format="URL", size=2) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=123, hash="18dnf8dfbakfdh", size=2, file_format="URL", lossless=True
        )

    def test_make_splash_url_when_no_hash(self, model):
        model.splash_hash = None
        assert model.make_splash_url(file_format="PNG", size=1024) is None

    def test_make_discovery_splash_url_when_hash(self, model):
        model.discovery_splash_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_GUILD_DISCOVERY_SPLASH", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_discovery_splash_url(file_format="URL", size=1024) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=123, hash="18dnf8dfbakfdh", size=1024, file_format="URL", lossless=True
        )

    def test_make_discovery_splash_url_when_no_hash(self, model):
        model.discovery_splash_hash = None
        assert model.make_discovery_splash_url(file_format="png", size=2048) is None

    def test_make_banner_url_when_hash(self, model):
        with mock.patch.object(
            routes, "CDN_GUILD_BANNER", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_banner_url(file_format="URL", size=512) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=123, hash="banner_hash", size=512, file_format="URL", lossless=True
        )

    def test_make_banner_url_when_format_is_None_and_banner_hash_is_for_animated(self, model):
        model.banner_hash = "a_18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_GUILD_BANNER", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_banner_url(file_format=None, size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=model.id, hash="a_18dnf8dfbakfdh", size=4096, file_format="GIF", lossless=True
        )

    def test_make_banner_url_when_format_is_None_and_banner_hash_is_not_for_animated(self, model):
        model.banner_hash = "18dnf8dfbakfdh"

        with mock.patch.object(
            routes, "CDN_GUILD_BANNER", new=mock.Mock(compile_to_file=mock.Mock(return_value="file"))
        ) as route:
            assert model.make_banner_url(file_format=None, size=4096) == "file"

        route.compile_to_file.assert_called_once_with(
            urls.CDN_URL, guild_id=model.id, hash=model.banner_hash, size=4096, file_format="PNG", lossless=True
        )

    def test_make_banner_url_when_no_hash(self, model):
        model.banner_hash = None
        assert model.make_banner_url(file_format="PNG", size=2048) is None

    @pytest.mark.asyncio
    async def test_fetch_owner(self, model):
        model.app.rest.fetch_member = mock.AsyncMock()

        assert await model.fetch_owner() is model.app.rest.fetch_member.return_value
        model.app.rest.fetch_member.assert_awaited_once_with(123, 1111)

    @pytest.mark.asyncio
    async def test_fetch_widget_channel(self, model):
        mock_channel = mock.Mock(channels_.GuildChannel)
        model.app.rest.fetch_channel = mock.AsyncMock(return_value=mock_channel)

        assert await model.fetch_widget_channel() is model.app.rest.fetch_channel.return_value
        model.app.rest.fetch_channel.assert_awaited_once_with(192729)

    @pytest.mark.asyncio
    async def test_fetch_widget_channel_when_None(self, model):
        model.widget_channel_id = None

        assert await model.fetch_widget_channel() is None

    @pytest.mark.asyncio
    async def test_fetch_rules_channel(self, model):
        mock_channel = mock.Mock(channels_.GuildTextChannel)
        model.app.rest.fetch_channel = mock.AsyncMock(return_value=mock_channel)

        assert await model.fetch_rules_channel() is model.app.rest.fetch_channel.return_value
        model.app.rest.fetch_channel.assert_awaited_once_with(123445)

    @pytest.mark.asyncio
    async def test_fetch_rules_channel_when_None(self, model):
        model.rules_channel_id = None

        assert await model.fetch_rules_channel() is None

    @pytest.mark.asyncio
    async def test_fetch_system_channel(self, model):
        mock_channel = mock.Mock(channels_.GuildTextChannel)
        model.app.rest.fetch_channel = mock.AsyncMock(return_value=mock_channel)

        assert await model.fetch_system_channel() is model.app.rest.fetch_channel.return_value
        model.app.rest.fetch_channel.assert_awaited_once_with(123888)

    @pytest.mark.asyncio
    async def test_fetch_system_channel_when_None(self, model):
        model.system_channel_id = None

        assert await model.fetch_system_channel() is None

    @pytest.mark.asyncio
    async def test_fetch_public_updates_channel(self, model):
        mock_channel = mock.Mock(channels_.GuildTextChannel)
        model.app.rest.fetch_channel = mock.AsyncMock(return_value=mock_channel)

        assert await model.fetch_public_updates_channel() is model.app.rest.fetch_channel.return_value
        model.app.rest.fetch_channel.assert_awaited_once_with(99699)

    @pytest.mark.asyncio
    async def test_fetch_public_updates_channel_when_None(self, model):
        model.public_updates_channel_id = None

        assert await model.fetch_public_updates_channel() is None

    @pytest.mark.asyncio
    async def test_fetch_afk_channel(self, model):
        mock_channel = mock.Mock(channels_.GuildVoiceChannel)
        model.app.rest.fetch_channel = mock.AsyncMock(return_value=mock_channel)

        assert await model.fetch_afk_channel() is model.app.rest.fetch_channel.return_value
        model.app.rest.fetch_channel.assert_awaited_once_with(1234)

    @pytest.mark.asyncio
    async def test_fetch_afk_channel_when_None(self, model):
        model.afk_channel_id = None

        assert await model.fetch_afk_channel() is None

    def test_get_channel(self, model):
        model.app.cache.get_guild_channel.return_value.guild_id = model.id
        assert model.get_channel(456) is model.app.cache.get_guild_channel.return_value
        model.app.cache.get_guild_channel.assert_called_once_with(456)

    def test_get_channel_when_not_from_guild(self, model):
        model.app.cache.get_guild_channel.return_value.guild_id = 654523123

        assert model.get_channel(456) is None

        model.app.cache.get_guild_channel.assert_called_once_with(456)

    def test_get_channel_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_channel(456) is None

    def test_get_member(self, model):
        assert model.get_member(456) is model.app.cache.get_member.return_value
        model.app.cache.get_member.assert_called_once_with(123, 456)

    def test_get_member_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_member(456) is None

    def test_get_presence(self, model):
        assert model.get_presence(456) is model.app.cache.get_presence.return_value
        model.app.cache.get_presence.assert_called_once_with(123, 456)

    def test_get_presence_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_presence(456) is None

    def test_get_voice_state(self, model):
        assert model.get_voice_state(456) is model.app.cache.get_voice_state.return_value
        model.app.cache.get_voice_state.assert_called_once_with(123, 456)

    def test_get_voice_state_when_no_cache_trait(self, model):
        model.app = object()
        assert model.get_voice_state(456) is None

    def test_get_my_member_when_not_shardaware(self, model):
        model.app = object()
        assert model.get_my_member() is None

    def test_get_my_member_when_no_me(self, model):
        model.app.get_me = mock.Mock(return_value=None)

        assert model.get_my_member() is None

        model.app.get_me.assert_called_once_with()

    def test_get_my_member(self, model):
        model.app.get_me = mock.Mock()
        model.app.get_me.return_value.id = 123

        with mock.patch.object(guilds.Guild, "get_member") as get_member:
            assert model.get_my_member() is get_member.return_value

        get_member.assert_called_once_with(123)
        model.app.get_me.assert_called_once_with()


class TestRestGuild:
    @pytest.fixture
    def model(self, mock_app):
        return guilds.RESTGuild(
            app=mock_app,
            id=snowflakes.Snowflake(123),
            splash_hash="splash_hash",
            discovery_splash_hash="discovery_splash_hash",
            banner_hash="banner_hash",
            icon_hash="icon_hash",
            features=[guilds.GuildFeature.ANIMATED_ICON],
            incidents=guilds.GuildIncidents(
                invites_disabled_until=None,
                dms_disabled_until=None,
                dm_spam_detected_at=datetime.datetime(2015, 5, 13, 1, 1, 1, 1, tzinfo=datetime.timezone.utc),
                raid_detected_at=None,
            ),
            name="some guild",
            application_id=snowflakes.Snowflake(9876),
            afk_channel_id=snowflakes.Snowflake(1234),
            afk_timeout=datetime.timedelta(seconds=60),
            default_message_notifications=guilds.GuildMessageNotificationsLevel.ONLY_MENTIONS,
            description=None,
            explicit_content_filter=guilds.GuildExplicitContentFilterLevel.ALL_MEMBERS,
            is_widget_enabled=False,
            max_video_channel_users=10,
            mfa_level=guilds.GuildMFALevel.NONE,
            owner_id=snowflakes.Snowflake(1111),
            preferred_locale="en-GB",
            premium_subscription_count=12,
            premium_tier=guilds.GuildPremiumTier.TIER_3,
            public_updates_channel_id=None,
            rules_channel_id=None,
            system_channel_id=None,
            vanity_url_code="yeet",
            verification_level=guilds.GuildVerificationLevel.VERY_HIGH,
            widget_channel_id=None,
            system_channel_flags=guilds.GuildSystemChannelFlag.SUPPRESS_PREMIUM_SUBSCRIPTION,
            emojis={},
            stickers={},
            roles={},
            approximate_active_member_count=1000,
            approximate_member_count=100,
            max_presences=100,
            max_members=100,
            nsfw_level=guilds.GuildNSFWLevel.AGE_RESTRICTED,
        )
